Train 2D CNN

2D Convolution을 기본 구성 요소로 하는 EEG classifier를 학습해보는 노트북.


환경 구성


Data Filtering by Diagnosis

Non-Vascular Dementia, Non-Vascular MCI, Normal


Configure the Train, Validation, and Test Splits

Split the filtered dataset and shuffle them

Wrap the splitted data using PyTorch Dataset

Data loader test

Train, validation, test dataloaders


Some useful functions for training


Train 2D CNN models

2D CNN: ResNet-like model

ResNet-18

ResNet-50